package com.future.interview;

import java.util.concurrent.atomic.AtomicInteger;

/**
 * Description: 面试题13. 机器人的运动范围
 * 地上有一个m行n列的方格，从坐标 [0,0] 到坐标 [m-1,n-1] 。一个机器人从坐标 [0, 0] 的格子开始移动，
 * 它每次可以向左、右、上、下移动一格（不能移动到方格外），也不能进入行坐标和列坐标的数位之和大于k的格子。
 * 例如，当k为18时，机器人能够进入方格 [35, 37] ，因为3+5+3+7=18。但它不能进入方格 [35, 38]，
 * 因为3+5+3+8=19。请问该机器人能够到达多少个格子？
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode.cn/problems/ji-qi-ren-de-yun-dong-fan-wei-lcof
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 *
 * @author weiruibai.vendor
 * Date: 2022/12/19 10:21
 */
public class Solution_13 {

    static Solution_13 solution = new Solution_13();

    public static void main(String[] args) {
        //int m = 6, n = 7, k = 5;
        int m = 38, n = 15, k = 9;
        System.out.println(solution.movingCount(m, n, k));
    }

    /**
     * 注意：要一步一步地走！！！，不能跳过，不能间断!!
     *
     * @param m
     * @param n
     * @param k
     * @return
     */
    public int movingCount(int m, int n, int k) {
        boolean[][] visited = new boolean[m][n];
        AtomicInteger atomic = new AtomicInteger(0);
        process(visited, 0, 0, atomic, k);
        /*for (int i = 0; i < m; i++) {
            if (getSum(i) > k) continue;
            for (int j = 0; j < n; j++) {
                if (visited[i][j]) {
                    System.out.print("(" + i + "," + j + ") ");
                }
            }
            System.out.println();
        }*/
        return atomic.get();
    }

    private void process(boolean[][] visited, int i, int j, AtomicInteger sum, int k) {
        int num1 = getSum(i);
        int num2 = getSum(j);
        if ((i < 0 || visited.length <= i) || (j < 0 || visited[0].length <= j) || visited[i][j] || num1 + num2 > k) {
            return;
        }
        //System.out.print("(" + i + "," + j + ") ");
        visited[i][j] = true;
        sum.incrementAndGet();
        // 上
        process(visited, i - 1, j, sum, k);
        // 下
        process(visited, i + 1, j, sum, k);
        // 左
        process(visited, i, j - 1, sum, k);
        // 右
        process(visited, i, j + 1, sum, k);
    }

    private int getSum(int num) {
        int sum = 0;
        while (num > 9) {
            sum += num % 10;
            num = num / 10;
        }
        sum += num;
        return sum;
    }
}
